Matplotlib radio button